/*
 *  Copyright (C) 2022-2024. Maximilian Keppeler (https://www.maxkeppeler.com)
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */
package com.maxkeppeler.sheets.test.utils

import androidx.compose.runtime.Composable
import androidx.compose.ui.test.junit4.ComposeContentTestRule
import androidx.compose.ui.test.junit4.ComposeTestRule
import androidx.compose.ui.test.onNodeWithTag
import com.maxkeppeker.sheets.core.utils.TestTags

fun ComposeContentTestRule.setContentAndWaitForIdle(content: @Composable () -> Unit) {
    setContent(content)
    waitForIdle()
}

fun ComposeTestRule.onDialog() = onNodeWithTag(TestTags.DIALOG_BASE_CONTENT)

fun ComposeTestRule.onPopup() = onNodeWithTag(TestTags.POPUP_BASE_CONTENT)

fun ComposeTestRule.onView() = onNodeWithTag(TestTags.FRAME_BASE_CONTENT)

fun ComposeTestRule.onExtraButton() = onNodeWithTag(TestTags.BUTTON_EXTRA)

fun ComposeTestRule.onPositiveButton() = onNodeWithTag(TestTags.BUTTON_POSITIVE)

fun ComposeTestRule.onNegativeButton() = onNodeWithTag(TestTags.BUTTON_NEGATIVE)

fun ComposeTestRule.onNodeWithTags(vararg testTag: Any, useUnmergedTree: Boolean = true) = onNodeWithTag(testTag.joinToString(), useUnmergedTree)

